from pages.basePage import BasePage
from pages.loginpage import loginPage
from pages.register import registerPage


class FirstPage(BasePage):
    def go_to_loginpage(self):
        if self.getElementValue(self.login_uname)=="登录":
            self.getElement(self.login_uname).click()
            return loginPage(self.wd)
    def go_to_registerpage(self):
        if self.getElementValue(self.register_loginout) == "注册":
            self.getElement(self.register_loginout).click()
            return registerPage(self.wd)

